# create_admin.py（生成管理员的脚本）
import os
# 初始化 Django 环境（必须加，否则找不到 User 模型）
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "ExamPulse.settings")  # 把“你的项目名”换成实际项目名
import django
django.setup()

from django.contrib.auth.models import User

# 配置管理员信息（自己改）
ADMIN_NAME = "admin"
ADMIN_PWD = "Admin123456"

if __name__ == "__main__":
    # 先检查管理员是否已存在，避免重复
    if not User.objects.filter(username=ADMIN_NAME).exists():
        # create_superuser 是 Django 专门生成管理员的方法
        User.objects.create_superuser(
            username=ADMIN_NAME,
            password=ADMIN_PWD,
            is_superuser=True  # 关键：标记为管理员
        )
        print(f"管理员 {ADMIN_NAME} 创建成功！密码：{ADMIN_PWD}")
    else:
        print(f"管理员 {ADMIN_NAME} 已经存在啦～")